package com.example.domain.repository;

import com.example.domain.entity.UnitConversionRule;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.math.BigDecimal;
import java.util.List;
import java.util.Optional;

/**
 * 单位换算规则Repository
 * 
 * @author Means
 * @since 2025-10-27
 */
@Repository
public interface UnitConversionRuleRepository extends JpaRepository<UnitConversionRule, String> {
    
    /**
     * 查询源单位到目标单位的换算规则
     * 
     * @param fromUnit 源单位
     * @param toUnit 目标单位
     * @return 换算规则
     */
    Optional<UnitConversionRule> findByFromUnitAndToUnit(String fromUnit, String toUnit);
    
    /**
     * 查询源单位到目标单位的启用换算规则
     * 
     * @param fromUnit 源单位
     * @param toUnit 目标单位
     * @param enabled 是否启用
     * @return 换算规则
     */
    Optional<UnitConversionRule> findByFromUnitAndToUnitAndEnabled(String fromUnit, String toUnit, Boolean enabled);
    
    /**
     * 查询某个单位作为源单位的所有换算规则
     * 
     * @param fromUnit 源单位
     * @return 换算规则列表
     */
    List<UnitConversionRule> findByFromUnit(String fromUnit);
    
    /**
     * 查询某个单位作为源单位的所有启用换算规则
     * 
     * @param fromUnit 源单位
     * @param enabled 是否启用
     * @return 换算规则列表
     */
    List<UnitConversionRule> findByFromUnitAndEnabled(String fromUnit, Boolean enabled);
    
    /**
     * 查询某个单位作为目标单位的所有换算规则
     * 
     * @param toUnit 目标单位
     * @return 换算规则列表
     */
    List<UnitConversionRule> findByToUnit(String toUnit);
    
    /**
     * 查询某个单位作为目标单位的所有启用换算规则
     * 
     * @param toUnit 目标单位
     * @param enabled 是否启用
     * @return 换算规则列表
     */
    List<UnitConversionRule> findByToUnitAndEnabled(String toUnit, Boolean enabled);
    
    /**
     * 查询所有启用的换算规则
     * 
     * @param enabled 是否启用
     * @return 换算规则列表
     */
    List<UnitConversionRule> findByEnabled(Boolean enabled);
    
    /**
     * 查询涉及某个单位的所有换算规则（作为源单位或目标单位）
     * 
     * @param unit 单位代码
     * @param enabled 是否启用
     * @return 换算规则列表
     */
    @Query("SELECT r FROM UnitConversionRule r WHERE (r.fromUnit = :unit OR r.toUnit = :unit) AND r.enabled = :enabled")
    List<UnitConversionRule> findByUnitAndEnabled(@Param("unit") String unit, @Param("enabled") Boolean enabled);
    
    /**
     * 获取换算系数
     * 
     * @param fromUnit 源单位
     * @param toUnit 目标单位
     * @return 换算系数
     */
    @Query("SELECT r.factor FROM UnitConversionRule r WHERE r.fromUnit = :fromUnit AND r.toUnit = :toUnit AND r.enabled = true")
    Optional<BigDecimal> getConversionFactor(@Param("fromUnit") String fromUnit, @Param("toUnit") String toUnit);
}

